import matplotlib
matplotlib.rcParams['backend'] = 'TkAgg'   # 或者 'Qt5Agg' 等其他后端
import matplotlib.pyplot as plt
import seaborn as sns

#  使用Seaborn内置的数据集
tips = sns.load_dataset('tips')

# 检查数据框的结构和数据类型
print(tips.head())
print(tips.dtypes)

# #  计算变量之间的相关性矩阵
correlation_matrix = tips.select_dtypes(include=['number']).corr()
# print(correlation_matrix)

#  创建热力图
sns.heatmap(correlation_matrix, annot=True, cmap='coolwarm')

#  添加标题
plt.title('Correlation  Heatmap')

#  显示图形
plt.show()
